import numpy as np

def NormalProb(energy, lamda):
    '''
    :param energy: 表示某个频带的能量
    :param lamda:  表示某个频带的方差
    :return: 概率值
    '''
    sumLamda = sum(lamda)
    prob = 1 / (np.pi * sumLamda) * np.exp(-(energy/sumLamda))
    return prob

def DiveProbe(energy, lamdaN):
    priSnr = energy / lamdaN
    postSnr = priSnr - 1
    diveRes = postSnr - np.log2(postSnr) - 1

    return diveRes

